/**
 * Copyright (c) 2018 人人开源 All rights reserved.
 * <p>
 * https://www.renren.io
 * <p>
 * 版权所有，侵权必究！
 */

package io.renren.datasource.config;

import java.util.ArrayDeque;
import java.util.Deque;

/**
 * 多数据源上下文
 *
 * @author Mark sunlightcs@gmail.com
 */
public class DynamicContextHolder {
  @SuppressWarnings("unchecked")
  private static final ThreadLocal<Deque<String>> CONTEXT_HOLDER = new ThreadLocal() {
    @Override
    protected Object initialValue() {
      return new ArrayDeque();
    }
  };

  /**
   * 获得当前线程数据源
   *
   * @return 数据源名称
   */
  public static String peek() {
    return CONTEXT_HOLDER.get().peek();
  }

  /**
   * 设置当前线程数据源
   *
   * @param dataSource 数据源名称
   */
  public static void push(String dataSource) {
    CONTEXT_HOLDER.get().push(dataSource);
  }

  /**
   * 清空当前线程数据源
   */
  public static void poll() {
    Deque<String> deque = CONTEXT_HOLDER.get();
    deque.poll();
    if (deque.isEmpty()) {
      CONTEXT_HOLDER.remove();
    }
  }

}